package com.mjf.utils;

import com.alibaba.fastjson.JSONObject;
import com.google.common.base.CaseFormat;
import com.mjf.common.GmallConfig;
import org.apache.commons.beanutils.BeanUtils;

import java.sql.*;
import java.util.ArrayList;
import java.util.List;

/**
 * Jdbc工具类
 */
public class JdbcUtil {

    /**
     * 根据指定 SQL 返回查询结果
     * @param conn 连接
     * @param sql 指定SQL（select * from t1;）
     * @param cls 查询结果对象
     * @param isRenameField 是否将字段名由下划线方式改为驼峰命名（order_id -> orderId）
     * @return [T(xx,xx,xx), T(xx,xx,xx)]
     */
    public static <T> List<T> queryList(Connection conn, String sql, Class<T> cls, boolean isRenameField) throws Exception {
        // 创建集合，用于存放查询结果数据
        ArrayList<T> resultList = new ArrayList<>();

        // 预编译SQL
        PreparedStatement preparedStatement = conn.prepareStatement(sql);

        // 执行查询
        ResultSet resultSet = preparedStatement.executeQuery();

        // 解析 resultSet
        ResultSetMetaData metaData = resultSet.getMetaData();
        int columnCount = metaData.getColumnCount();
        while (resultSet.next()) {
            // 创建泛型对象
            T t = cls.newInstance();

            // 给泛型对象赋值
            for (int i = 1; i < columnCount + 1; i++) {
                // 获取列名
                String columnName = metaData.getColumnName(i);
                // 判断是否需要转换为驼峰命名 -- 使用谷歌工具类
                if(isRenameField) {
                    columnName = CaseFormat.LOWER_UNDERSCORE.to(CaseFormat.LOWER_CAMEL, columnName.toLowerCase());
                }

                // 获取列值
                Object columnValue = resultSet.getObject(i);

                // 给泛型对象赋值 -- 使用谷歌工具类
                BeanUtils.setProperty(t, columnName, columnValue);
            }

            // 将该对象添加至集合
            resultList.add(t);
        }

        // 关闭资源
        preparedStatement.close();
        resultSet.close();

        return resultList;
    }

    public static void main(String[] args) throws Exception {

//        System.out.println(CaseFormat.LOWER_UNDERSCORE.to(CaseFormat.LOWER_CAMEL, "order_info"));

        Class.forName(GmallConfig.PHOENIX_DRIVER);
        Connection conn = DriverManager.getConnection(GmallConfig.PHOENIX_SERVER);
        List<JSONObject> queryList = queryList(conn, "select * from GMALL_REALTIME.DIM_USER_INFO", JSONObject.class, true);

        for (JSONObject jsonObject : queryList) {
            System.out.println(jsonObject);
        }

        conn.close();

    }

}
